package org.zecontech.extraplus.datagen;

import net.minecraft.core.HolderLookup;
import net.minecraft.data.PackOutput;
import net.minecraft.data.tags.BiomeTagsProvider;
import net.minecraft.tags.BiomeTags;
import org.zecontech.extraplus.ExtraPlus;
import org.zecontech.extraplus.datagen.levelgen.ModBiomeProvider;

import java.util.concurrent.CompletableFuture;

/**
 * 生物群系标签提供者
 */
public class ModBiomeTagsProvider extends BiomeTagsProvider {

    public ModBiomeTagsProvider(PackOutput output, CompletableFuture<HolderLookup.Provider> provider) {
        super(output, provider, ExtraPlus.MOD_ID);
    }

    /**
     * 生物群系标签
     * @param provider 提供者
     */
    @Override
    protected void addTags(HolderLookup.Provider provider) {
        tag(BiomeTags.IS_OVERWORLD).addOptional(ModBiomeProvider.WILLOW_FOREST);
        tag(BiomeTags.IS_FOREST).addOptional(ModBiomeProvider.WILLOW_FOREST);
    }
}
